# coding: utf-8
"""
@Time    : 2024/8/14 9:47
@Author  : Y.H LEE
"""

from utils.tools import *
from sys_params import *
# from minibatch_iterator import MinibatchIterator
from grid_search import GridSearch

'''
Model main function
'''


def main():
    """set Random seed"""
    set_seed(123456, device)

    """load params and grid search params"""
    hp_params = load_json(h_params)
    sc_params = load_json(search_params)

    """model train"""
    GridSearch(hp_params, sc_params).train_with_hyperparams_grid_search(save_model=True)
    # trainer(minibatch)


if __name__ == '__main__':
    main()
